square of a diagonal matrix